diff --git a/drivers/demo/demo-shape.c b/drivers/demo/demo-shape.c
index 8686c99..0f5cda2 100644
--- a/drivers/demo/demo-shape.c
+++ b/drivers/demo/demo-shape.c
@@ -193,6 +193,6 @@
 	.ops	= &shape_ops,
 	.probe = dm_shape_probe,
 	.remove = dm_shape_remove,
-	.priv_auto_alloc_size = sizeof(struct shape_data),
-	.platdata_auto_alloc_size = sizeof(struct dm_demo_pdata),
+	.priv_auto	= sizeof(struct shape_data),
+	.platdata_auto	= sizeof(struct dm_demo_pdata),
 };